package LC_81_100;

/**
 * 给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。
 * 你应当保留两个分区中每个节点的初始相对位置。
 * <p>
 * 示例:
 * 输入: head = 1->4->3->2->5->2, x = 3
 * 输出: 1->2->2->4->3->5
 */
public class LC86 {
    public ListNode partition(ListNode head, int x) {
        ListNode smallHead = new ListNode(-1);
        ListNode greatHead = new ListNode(-1);
        ListNode smallCur = smallHead;
        ListNode greatCur = greatHead;
        ListNode cur = head;
        while (cur != null) {
            if (cur.val < x) {
                smallCur.next = cur;
                smallCur = smallCur.next;
            } else {
                greatCur.next = cur;
                greatCur = greatCur.next;
            }
            cur = cur.next;
        }
        smallCur.next = greatHead.next;
        //由于greatCur是合并链表的最后一个节点，所以它要指向null
        greatCur.next = null;
        return smallHead.next;
    }
}
